Gao, Yanjie, Liu, Yu, Zhang, Hongyu, Li, Zhengxian, Zhu, Yonghao, Lin, Haoxiang, Yang, Mao. Association for Computing Machinery; 2020. Estimating GPU memory consumption of deep learning models.

Gao, Yanjie, Li, Zhengxian, Lin, Haoxiang, Zhang, Hongyu, Wu, Ming, Yang, Mao. Institute of Electrical and Electronics Engineers (IEEE); 2022. REFTY: Refinement Types for Valid Deep Learning Models.